#!/usr/bin/env python
"""
清空会话数据的辅助脚本
运行此脚本后再执行迁移
"""
import os
import sys
import django

# 设置Django环境
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'deepseek_project.settings')
django.setup()

from deepseek_api.models import ConversationSession

# 清空所有会话
count = ConversationSession.objects.count()
ConversationSession.objects.all().delete()
print(f"✅ 已清空 {count} 条会话记录")
print("现在可以运行: python3 manage.py migrate")
